package com.javayz.fileuploadanddownload.util;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.util.Arrays;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;


public class ZipUtils {

    private static final int BUFFER_SIZE = 2 * 4056;

    /**
     * 压缩成ZIP方法
     *
     * @param srcDir           压缩文件夹路径
     * @param out              压缩文件输出流
     * @param KeepDirStructure 是否保留原来的目录结构,true:保留目录结构;
     * @throws RuntimeException 压缩失败会抛出运行时异常
     * @false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)
     */
    public static void toZip(String srcDir,
                             OutputStream out,
                             boolean KeepDirStructure) {

        try {
            ZipOutputStream zos = new ZipOutputStream(out);
            File sourceFile = new File(srcDir);
            compress(sourceFile, zos, sourceFile.getName(), KeepDirStructure);
            zos.close();
        } catch (Exception e) {
            throw new RuntimeException("zip error from ZipUtils", e);
        }
    }

    /**
     * 递归压缩方法
     *
     * @param sourceFile       源文件
     * @param zos              zip输出流
     * @param name             压缩后的名称
     * @param KeepDirStructure 是否保留原来的目录结构,true:保留目录结构;
     * @throws Exception
     * @false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)
     */
    private static void compress(File sourceFile,
                                 ZipOutputStream zos,
                                 String name,
                                 boolean KeepDirStructure) {

        byte[] buf = new byte[BUFFER_SIZE];
        if (sourceFile.isFile()) {
            try {
                // 向zip输出流中添加一个zip实体，构造器中name为zip实体的文件的名字
                zos.putNextEntry(new ZipEntry(name));
                // copy文件到zip输出流中
                int len;
                FileInputStream in = new FileInputStream(sourceFile);
                while ((len = in.read(buf)) != -1) {
                    zos.write(buf, 0, len);
                }
                // Complete the entry
                zos.closeEntry();
                in.close();
            } catch (IOException e) {
                e.printStackTrace();
            }

        } else {
            File[] listFiles = sourceFile.listFiles();
            if (listFiles == null || listFiles.length == 0) {
                // 需要保留原来的文件结构时,需要对空文件夹进行处理
                if (KeepDirStructure) {
                    try {
                        // 空文件夹的处理
                        zos.putNextEntry(new ZipEntry(name + "/"));
                        // 没有文件，不需要文件的copy
                        zos.closeEntry();
                        zos.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            } else {
                for (File file : listFiles) {
                    // 判断是否需要保留原来的文件结构
                    if (KeepDirStructure) {
                        // 注意：file.getName()前面需要带上父文件夹的名字加一斜杠,
                        // 不然最后压缩包中就不能保留原来的文件结构,即：所有文件都跑到压缩包根目录下了
                        compress(file, zos, name + "/" + file.getName(), KeepDirStructure);
                    } else {
                        compress(file, zos, file.getName(), KeepDirStructure);
                    }
                }
            }
        }
    }

    /**
     * @描述：文件下载
     * @参数： [response, urlFile]
     * @返回值： void
     * @创建人： ZLQ
     * @创建时间： 2022/11/18 15:27
     * @修改人和其它信息：
     */
    public static void downFile(HttpServletResponse response,
                                String urlFile) {

        try {

            File file = new File(urlFile);
            String filename = file.getName();
            // String fileType = filename.substring(filename.indexOf(".") + 1);

            // 以流的形式下载文件
            FileInputStream fileInputStream = new FileInputStream(file);
            InputStream fis = new BufferedInputStream(fileInputStream);
            byte[] buffer = new byte[1024];
            int len = 0;
            // 清空response
            response.reset();
            // 设置response的Header
            // 解决跨域
            response.addHeader("Access-Control-Allow-Origin", "*");
            boolean b = false;
            if (b) {

            } else {
                // 直接下载
                response.setContentType("application/text;charset=utf-8");
                response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filename, "UTF-8"));
                response.addHeader("Content-Length", "" + file.length());
            }
            OutputStream toClient = new BufferedOutputStream(response.getOutputStream());
            int a = 0;
            while ((len = fis.read(buffer)) != -1) {
                a = len + a;
                toClient.write(buffer, 0, len);
            }
            fileInputStream.close();
            fis.close();
            toClient.flush();
            toClient.close();
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

    /**
     * @描述：删除文件夹方法
     * @参数： [dirPath]
     * @返回值： void
     * @创建人： ZLQ
     * @创建时间： 2022/11/10 14:26
     * @修改人和其它信息：
     */
    public static void deleteDir(File directory) {

        if (!directory.isDirectory()) {
            directory.delete();
        } else {
            File[] files = directory.listFiles();
            // 空文件夹
            if (files.length == 0) {
                directory.delete();
                return;
            }

            // 删除子文件夹和子文件
            for (File file : files) {
                if (file.isDirectory()) {
                    deleteDir(file);
                } else {
                    file.delete();
                }
            }

            // 删除文件夹本身
            directory.delete();
        }

    }

    /**
     * @描述：删除压缩包方法
     * @参数： [filePath]
     * @返回值： void
     * @创建人： ZLQ
     * @创建时间： 2022/11/10 14:26
     * @修改人和其它信息：
     */
    public static void deleteFile(String filePath) {

        File file = new File(filePath);
        if (!file.exists()) {
            return;
        }
        if (file.isDirectory()) {
            File[] list = file.listFiles();

            for (File f : list) {
                deleteFile(f.getAbsolutePath());
            }
        }
        file.delete();
    }

    /*public static void main(String[] args) {
        FixFileName("D:/demo/工程基本信息/sss/1001N1100000000299YJ/321wer21","D:/demo/工程基本信息/sss/1001N1100000000299YJ/36e07ae95ff0411cad973cf7449d0daf..xlsx");
    }*/
    public static void FixFileName(String filePath, String newFileName) {
        // 旧的文件或目录
        File oldName = new File(filePath);
        // 新的文件或目录
        File newName = new File(newFileName);
        if (newName.exists()) {  //  确保新的文件名不存在
            System.out.println("新文件名已存在");
        }
        if (oldName.renameTo(newName)) {
            System.out.println("已重命名");
        } else {
            System.out.println("Error");
        }
    }


}
